# -*- coding: utf-8 -*-
# @Time: 2025/6/11 0:48
# @Author: wzd
# @Email: 2146333089@qq.com
# @File: merge_pdf.py
import os
from PyPDF2 import PdfMerger


def merge_pdfs_in_directory(directory_path, output_path="merged.pdf"):
    """合并指定目录下的所有PDF文件"""
    # 获取目录中所有PDF文件
    pdf_files = [f for f in os.listdir(directory_path) if f.lower().endswith('.pdf')]

    # 按文件名排序（可选）
    pdf_files.sort()

    # 创建合并器对象
    merger = PdfMerger()

    # 遍历并添加所有PDF文件
    for pdf_file in pdf_files:
        file_path = os.path.join(directory_path, pdf_file)
        print(f"正在添加: {pdf_file}")
        merger.append(file_path)

    # 写入合并后的文件
    merger.write(output_path)
    merger.close()
    print(f"合并完成！文件已保存至: {output_path}")


# 使用示例
if __name__ == "__main__":
    # 指定包含PDF文件的目录路径
    pdf_directory = r"C:\downloads\v_pdf"  # 替换为你的PDF目录路径

    # 调用函数合并PDF
    merge_pdfs_in_directory(pdf_directory, r"C:\downloads\v_pdf\All_pdfs_merged.pdf")
